Adding Immunity To An Infectious Disease Model